package com.sanli.swing.header;

import java.awt.Color;
import java.awt.Component;
import java.awt.Dimension;
import java.awt.Graphics;
import java.awt.Rectangle;
import java.util.Enumeration;
import java.util.Hashtable;

import javax.swing.JComponent;
import javax.swing.JLabel;
import javax.swing.JTable;
import javax.swing.UIManager;
import javax.swing.plaf.basic.BasicTableHeaderUI;
import javax.swing.table.DefaultTableCellRenderer;
import javax.swing.table.TableCellRenderer;
import javax.swing.table.TableColumn;
import javax.swing.table.TableColumnModel;


public class GroupableTableHeaderUI extends BasicTableHeaderUI {
	  
	  public void paint(Graphics g, JComponent c) {
	    Rectangle clipBounds = g.getClipBounds();
	    if (header.getColumnModel() == null) return;
	    ((GroupableTableHeader)header).setColumnMargin();
	    int column = 0;
	    Dimension size = header.getSize();
	    Rectangle cellRect  = new Rectangle(0, 0, size.width, size.height);
	    Hashtable h = new Hashtable();
	    int columnMargin = header.getColumnModel().getColumnMargin();
	    
	    Enumeration enumeration = header.getColumnModel().getColumns();
	    while (enumeration.hasMoreElements()) {
	      cellRect.height = size.height;
	      cellRect.y      = 0;
	      TableColumn aColumn = (TableColumn)enumeration.nextElement();
	      Enumeration cGroups = ((GroupableTableHeader)header).getColumnGroups(aColumn);
	      if (cGroups != null) {
	        int groupHeight = 0;
	        while (cGroups.hasMoreElements()) {
	          ColumnGroup cGroup = (ColumnGroup)cGroups.nextElement();
	          Rectangle groupRect = (Rectangle)h.get(cGroup);
	          if (groupRect == null) {
	            groupRect = new Rectangle(cellRect);
	            Dimension d = cGroup.getSize(header.getTable());
	            groupRect.width  = d.width;
	            groupRect.height = d.height;    
	            h.put(cGroup, groupRect);
	          }
	          paintCell(g, groupRect, cGroup);
	          groupHeight += groupRect.height;
	          cellRect.height = size.height - groupHeight;
	          cellRect.y      = groupHeight;
	        }
	      }      
	      cellRect.width = aColumn.getWidth() + 0;
	      if (cellRect.intersects(clipBounds)) {
	        paintCell(g, cellRect, column);
	      }
	      cellRect.x += cellRect.width;
	      column++;
	    }
	  }

	  private void paintCell(Graphics g, Rectangle cellRect, int columnIndex) {
	    TableColumn aColumn = header.getColumnModel().getColumn(columnIndex);
	    TableCellRenderer renderer = aColumn.getHeaderRenderer();
	    //revised by Java2s.com
	    renderer = new DefaultTableCellRenderer(){
	        public Component getTableCellRendererComponent(JTable table, Object value, boolean isSelected, boolean hasFocus, int row, int column) {
	          JLabel header = new JLabel();
	            header.setForeground(table.getTableHeader().getForeground());
	            header.setBackground(table.getTableHeader().getBackground());
	            header.setFont(table.getTableHeader().getFont());

	          header.setHorizontalAlignment(JLabel.CENTER);
	          header.setText(value.toString());
	          header.setBorder(UIManager.getBorder("TableHeader.cellBorder"));
	            return header;
	        }
	    
	    };
	    Component c = renderer.getTableCellRendererComponent(
	        header.getTable(), aColumn.getHeaderValue(),false, false, -1, columnIndex);
	        
	        c.setBackground(UIManager.getColor("control"));
//	        c.setBackground(Color.red);
	      
	    rendererPane.add(c);
	    rendererPane.paintComponent(g, c, header, cellRect.x, cellRect.y,
	        cellRect.width, cellRect.height, true);
	  }

	  private void paintCell(Graphics g, Rectangle cellRect,ColumnGroup cGroup) {
	    TableCellRenderer renderer = cGroup.getHeaderRenderer();
	      //revised by Java2s.com
	     // if(renderer == null){
//	      return ;
	  //    }

	    Component component = renderer.getTableCellRendererComponent(
	      header.getTable(), cGroup.getHeaderValue(),false, false, -1, -1);
	    rendererPane.add(component);
	    rendererPane.paintComponent(g, component, header, cellRect.x, cellRect.y,
	        cellRect.width, cellRect.height, true);
	  }

	  private int getHeaderHeight() {
	    int height = 0;
	    TableColumnModel columnModel = header.getColumnModel();
	    for(int column = 0; column < columnModel.getColumnCount(); column++) {
	      TableColumn aColumn = columnModel.getColumn(column);
	      TableCellRenderer renderer = aColumn.getHeaderRenderer();
	      //revised by Java2s.com
	      if(renderer == null){
	    	  return 60;
	      }
	      
	      Component comp = renderer.getTableCellRendererComponent(
	        header.getTable(), aColumn.getHeaderValue(), false, false,-1, column);
	      int cHeight = comp.getPreferredSize().height;
	      Enumeration e = ((GroupableTableHeader)header).getColumnGroups(aColumn);      
	      if (e != null) {
	        while (e.hasMoreElements()) {
	          ColumnGroup cGroup = (ColumnGroup)e.nextElement();
	          cHeight += cGroup.getSize(header.getTable()).height;
	        }
	      }
	      height = Math.max(height, cHeight);
	    }
	    return height;
	  }

	  private Dimension createHeaderSize(long width) {
	    TableColumnModel columnModel = header.getColumnModel();
	    width += columnModel.getColumnMargin() * columnModel.getColumnCount();
	    if (width > Integer.MAX_VALUE) {
	      width = Integer.MAX_VALUE;
	    }
	    return new Dimension((int)width, getHeaderHeight());
	  }

	  public Dimension getPreferredSize(JComponent c) {
	    long width = 0;
	    Enumeration enumeration = header.getColumnModel().getColumns();
	    while (enumeration.hasMoreElements()) {
	      TableColumn aColumn = (TableColumn)enumeration.nextElement();
	      width = width + aColumn.getPreferredWidth();
	    }
	    return createHeaderSize(width);
	  }
	}

